load(
    "//tensorflow/core/platform:default/build_config.bzl",
    "tf_proto_library",
    "tf_pyclif_proto_library",
)

package(
    default_visibility = ["//visibility:public"],
    licenses = ["notice"],  # Apache 2.0
)

exports_files([
    "LICENSE",
    "generic_tree_model_proto.swig",
])

tf_proto_library(
    name = "generic_tree_model",
    srcs = ["generic_tree_model.proto"],
    cc_api_version = 2,
    visibility = ["//visibility:public"],
)

tf_proto_library(
    name = "generic_tree_model_extensions",
    srcs = ["generic_tree_model_extensions.proto"],
    cc_api_version = 2,
    protodeps = [":generic_tree_model"],
    visibility = ["//visibility:public"],
)

tf_pyclif_proto_library(
    name = "generic_tree_model_pyclif",
    proto_lib = ":generic_tree_model",
    proto_srcfile = "generic_tree_model.proto",
    visibility = ["//visibility:public"],
)
